python gen algorithm